import { SimpleRag } from "./demo";

/**
 * 主函数
 * @description 项目的入口函数,测试一下封装的实际的大模型
 * @author RAG Demo Project
 * @version 1.0.0
*/
async function main() {
  const rag = new SimpleRag();
  await rag.initialize();
  const inserted = await rag.add("RAG 技术原理");
  console.log(JSON.stringify(inserted));
  const res = await rag.query("RAG 是什么？");
  console.log(JSON.stringify(res));
  const res2 = await rag.query("随便问一些无关内容");
  console.log(JSON.stringify(res2));
  await rag.del(inserted);
}

main();
